package com.tbryant.test.redis.usagescenarios;

import java.util.Set;

import com.tbryant.test.redis.RedisConstant;
import redis.clients.jedis.Jedis;

/**
 * 使用有序集合实现延时队列
 */
public class DelayQueue {

    private static final Jedis JEDIS = new Jedis(RedisConstant.REDIS_IP, RedisConstant.REDIS_PORT);

    /**
     * 推送消息
     * @param delaySecond 延迟时间（秒）
     */
    static Long setMessage(String queueKey, String message, int delaySecond) {
        return JEDIS.zadd(queueKey, System.currentTimeMillis() + delaySecond * 1000, message);
    }

    /**
     * 拉取一条消息
     * 暂时没有消息返回null
     */
    static String getMessage(String queueKey) {
        Set<String> values = JEDIS.zrangeByScore(queueKey, 0, System.currentTimeMillis(), 0, 1);
        if (!values.isEmpty()) {
            String message = values.iterator().next();
            System.out.println(message);
            // 竞争这条数据
            if (JEDIS.zrem(queueKey, message) > 0) {
                return message;
            }
        }
        return null;
    }
}